from enum import unique

from myapp.const.base import TEnum
from myapp.const.mq import SlowTopicPrefix

# Message queue key for when a dataset augmentation task is created.
MsgKeyDatasetAugTaskCreated = SlowTopicPrefix + "dataset_aug_task_created"
# Message queue key for an inference task within dataset augmentation.
MsgKeyDatasetAugTaskInfer = SlowTopicPrefix + "dataset_aug_task_infer"
# Message queue key for when an inference task within dataset augmentation is completed.
MsgKeyDatasetAugTaskInferDone = SlowTopicPrefix + "dataset_aug_task_infer_done"


# Defines an enumeration for various dataset augmentation statuses.
@unique
class EnumDatasetAugStatus(TEnum):
    """
    数据增强状态
    """
    # Indicates that the dataset augmentation task is pending or queued.
    pending = "pending"  # 排队中
    # Indicates that the dataset augmentation task is currently being processed.
    processing = "processing"  # 处理中
    # Indicates that the dataset augmentation task has successfully completed.
    succeed = "succeed"  # 导入完成
    # Indicates that the dataset augmentation task has failed.
    failed = "failed"  # 失败